to_drop = {}

to_drop['origin_importance_0'] = {'user_maxprice_1month',  # importance = 0
                                  'user_roomservice_4_5ratio_3month',
                                  'user_roomservice_7_0ratio_1month',
                                  'user_roomservice_4_5ratio_1month',
                                  'user_roomservice_4_4ratio_1month',
                                  'user_roomservice_4_3ratio_1month',
                                  'user_roomservice_4_0ratio_1month',
                                  'user_ordnum_3month',
                                  'user_roomservice_7_0ratio_3month',
                                  'user_roomservice_4_0ratio_1week',
                                  'user_roomservice_4_4ratio_3month',
                                  'user_roomservice_4_2ratio_3month',
                                  'user_roomservice_4_3ratio_3month',
                                  'hotel_minprice_lastord',
                                  'user_ordnum_1month',
                                  'roomtag_2',
                                  'user_roomservice_4_3ratio_1week',
                                  'ordertype_9_ratio',
                                  'orderbehavior_7_ratio',
                                  'orderbehavior_4_ratio_3month',
                                  'orderbehavior_5_ratio_1week',
                                  'orderbehavior_4_ratio_1week',
                                  'orderbehavior_1_ratio',
                                  'user_avgdealprice',
                                  'ordertype_11_ratio',
                                  'ordertype_8_ratio',
                                  'user_ordernum',
                                  'ordertype_5_ratio',
                                  'ordertype_4_ratio',
                                  'ordertype_3_ratio',
                                  'ordertype_2_ratio',
                                  'ordertype_1_ratio',
                                  'user_avgrecommendlevel',
                                  'user_confirmtime',
                                  'orderbehavior_9',
                                  'user_activation',
                                  'user_roomservice_4_2ratio_1week',
                                  'user_roomservice_7_0ratio',
                                  'user_roomservice_4_4ratio_1week',
                                  'user_roomservice_4_5ratio_1week',
                                  'user_roomservice_7_0ratio_1week',
                                  'user_roomservice_3_123ratio_1week',
                                  'user_avgprice_1week',
                                  'user_ordnum_1week',
                                  'user_roomservice_8_1ratio',
                                  'user_roomservice_6_2ratio',
                                  'user_stdprice',
                                  'user_roomservice_4_1ratio',
                                  'user_roomservice_4_4ratio',
                                  'user_roomservice_4_3ratio',
                                  'user_roomservice_4_2ratio',
                                  'user_roomservice_4_0ratio',
                                  'user_avgroomarea',
                                  'user_citynum',
                                  'user_roomservice_6_1ratio'}

to_drop['origin_pearsonr>=0.99'] = {'basic_minprice_lastord',  # pearsonr >=0.99
                                    'roomtag_4_lastord_nan',
                                    'roomtag_6_lastord_nan',
                                    'star_lastord_nan',
                                    'roomservice_5_1.0',
                                    'basic_maxarea',
                                    'roomservice_7_1.0',
                                    'roomservice_2_1.0',
                                    'roomtag_4_1.0',
                                    'roomtag_5_lastord_nan',
                                    'roomservice_3_lastord_nan',
                                    'roomservice_8_lastord_nan',
                                    'roomtag_6_lastord_0.0',
                                    'roomtag_5_1.0',
                                    'roomservice_5_lastord_nan',
                                    'roomtag_1_1.0'}

to_drop['origin_pearsonr>=0.95'] = {'star_lastord_nan',
                                    'roomtag_5_lastord_0.0',
                                    'roomservice_2_1.0',
                                    'roomtag_6_lastord_nan',
                                    'roomservice_3_lastord_nan',
                                    'basic_minprice_lastord',
                                    'basic_maxarea',
                                    'user_medprice_3month',
                                    'roomtag_5_lastord_nan',
                                    'roomservice_5_lastord_nan',
                                    'roomtag_6_lastord_0.0',
                                    'roomtag_4_lastord_nan',
                                    'roomservice_5_1.0',
                                    'roomtag_4_1.0',
                                    'roomtag_5_1.0',
                                    'roomservice_8_lastord_nan',
                                    'user_medprice_1month',
                                    'roomservice_4_lastord_nan',
                                    'roomtag_1_1.0',
                                    'roomservice_7_1.0',
                                    'roomservice_6_lastord_nan'}


def get_all_to_drop():
    result = set()
    for key in to_drop:
        print(key)
        result = result.union(to_drop[key])
    return result
